package com.xiatl.cgi.server.cgi.utils;


import com.alibaba.fastjson.JSONObject;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.net.HttpURLConnection;
import java.net.URL;
import java.net.URLConnection;
import java.util.Map;

/**
 * Http请求工具类
 */
public class HttpRequestUtil {

    /**
     * get请求
     *
     * @param url 请求url
     * @return 返回结果
     */
    public static String get(String url, Map<String, String> headers) {
        StringBuilder stringBuilder = new StringBuilder();
        BufferedReader in = null;
        try {

            URL realUrl = new URL(url);
            // 打开和URL之间的连接
            URLConnection connection = realUrl.openConnection();
            // 设置通用的请求属性
            connection.setRequestProperty("accept", "*/*");
            connection.setRequestProperty("connection", "Keep-Alive");
            connection.setRequestProperty("user-agent", "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1;SV1)");
            // 设置header
            if (headers != null && headers.size() > 0) {
                for (Map.Entry<String, String> map : headers.entrySet()) {
                    connection.setRequestProperty(map.getKey(), map.getValue());
                }
            }
            // 建立实际的连接
            connection.connect();
            // 定义 BufferedReader输入流来读取URL的响应
            in = new BufferedReader(new InputStreamReader(connection.getInputStream()));
            String line;
            while ((line = in.readLine()) != null) {
                stringBuilder.append(line);
            }
            return stringBuilder.toString();
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }
        // 使用finally块来关闭输入流
        finally {
            try {
                if (in != null) {
                    in.close();
                }
            } catch (Exception e2) {
                e2.printStackTrace();
            }
        }

    }

    /**
     * post请求
     *
     * @param url    url
     * @param params 参数
     * @return 结果
     */
    public static String post(String url, JSONObject params, Map<String, String> headers) {
        return request(url, params, "post", headers);
    }

    /**
     * 请求
     *
     * @param url    url
     * @param params 参数
     * @param method 请求方式
     * @return 结果
     */
    private static String request(String url, JSONObject params, String method, Map<String, String> headers) {
        PrintWriter out = null;
        BufferedReader in = null;
        StringBuilder stringBuilder = new StringBuilder();
        // 我们这合理明确url是带参数的
        if (!"post".equals(method)) {
            url = url + "&_method=" + method;
        }
        try {
            URL realUrl = new URL(url);
            // 打开和URL之间的连接
            HttpURLConnection conn = (HttpURLConnection) realUrl.openConnection();
            // 设置通用的请求属性
            conn.setRequestProperty("Content-Type", "application/json; charset=utf-8");
            conn.setRequestProperty("accept", "*/*");
            conn.setRequestProperty("connection", "Keep-Alive");
            conn.setRequestProperty("user-agent", "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1;SV1)");
            // 设置header
            if (headers != null && headers.size() > 0) {
                for (Map.Entry<String, String> map : headers.entrySet()) {
                    conn.setRequestProperty(map.getKey(), map.getValue());
                }
            }
            // 发送POST请求必须设置如下两行
            conn.setDoOutput(true);
            conn.setDoInput(true);
            // 获取URLConnection对象对应的输出流
            out = new PrintWriter(conn.getOutputStream());
            // 发送请求参数
            if (params != null)
                out.print(params.toString());
            // flush输出流的缓冲
            out.flush();
            // 定义BufferedReader输入流来读取URL的响应
            in = new BufferedReader(new InputStreamReader(conn.getInputStream()));
            String line;
            while ((line = in.readLine()) != null) {
                stringBuilder.append(line);
            }
            return stringBuilder.toString();
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }
        // 使用finally块来关闭输出流、输入流
        finally {
            try {
                if (out != null)
                    out.close();
                if (in != null)
                    in.close();
            } catch (IOException ex) {
                ex.printStackTrace();
            }
        }
    }

    /**
     * delete请求
     *
     * @param url    url
     * @param params 参数
     * @return 结果
     */
    public static String delete(String url, JSONObject params, Map<String, String> headers) {
        return request(url, params, "DELETE", headers);
    }

    /**
     * patch请求
     *
     * @param url    url
     * @param params 参数
     * @return 结果
     */
    public static String patch(String url, JSONObject params, Map<String, String> headers) {
        return request(url, params, "patch", headers);
    }


}
